package org.fujay.commons.langchain4j.core.enumd;

import lombok.AllArgsConstructor;
import lombok.Getter;

import java.util.List;

/**
 * @author slm
 * @description AI 模型厂商枚举
 */
@Getter
@AllArgsConstructor
public enum ModelSupplierEnum {

    OPEN_AI("1", "openAi", List.of(SupportTypeEnum.TEXT, SupportTypeEnum.IMAGE, SupportTypeEnum.VIDEO)),
    OLLAMA("2", "ollama", List.of(SupportTypeEnum.TEXT, SupportTypeEnum.IMAGE)),
    LOCAL_AI("3", "localAi", List.of(SupportTypeEnum.TEXT, SupportTypeEnum.IMAGE)),
    DEFAULT_AI("9", "defaultAi", List.of());

    private final String code;

    private final String name;
    /**
     * 模型支持的输入类型
     */
    private final List<SupportTypeEnum> supportTypes;
}
